package com.mingdao.lhy.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 
 * @author chenyj
 * @since 2018/12/03
 * 
 */
public class ReadExcel {

	private static Logger log = LoggerFactory.getLogger(ReadExcel.class);

	// 总行数
	private int totalRows = 0;
	// 总条数
	private int totalCells = 0;
	// 错误信息接收器
	private String errorMsg;

	// 构造方法
	public ReadExcel() {
	}

	// 获取总行数
	public int getTotalRows() {
		return totalRows;
	}

	// 获取总列数
	public int getTotalCells() {
		return totalCells;
	}

	// 获取错误信息
	public String getErrorInfo() {
		return errorMsg;
	}

	/**
	 * 读EXCEL文件，获取信息集合
	 * 
	 * @param fielName
	 * @return
	 */
	public List<Map<String, Object>> getExcelInfo(String fileName) {
		// String fileName = mFile.getOriginalFilename();// 获取文件名
		// List<Map<String, Object>> userList = new LinkedList<Map<String, Object>>();
		try {
			if (!validateExcel(fileName)) {// 验证文件名是否合格
				return null;
			}
			boolean isExcel2003 = true;// 根据文件名判断文件是2003版本还是2007版本
			if (isExcel2007(fileName)) {
				isExcel2003 = false;
			}
			File f = new File(fileName);

			return createExcel((new FileInputStream(f)), isExcel2003);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;
	}

	/**
	 * 根据excel里面的内容读取客户信息
	 * 
	 * @param is          输入流
	 * @param isExcel2003 excel是2003还是2007版本
	 * @return
	 * @throws IOException
	 */
	public List<Map<String, Object>> createExcel(InputStream is, boolean isExcel2003) {
		try {
			Workbook wb = null;
			if (isExcel2003) {// 当excel是2003时,创建excel2003
				wb = new HSSFWorkbook(is);
			} else {// 当excel是2007时,创建excel2007
				wb = new XSSFWorkbook(is);
			}
			return readExcelValue(wb);// 读取Excel里面客户的信息
		} catch (IOException e) {
			e.printStackTrace();
		}
		return null;
	}

	/**
	 * 读取Excel里面客户的信息
	 * 
	 * @param wb
	 * @return
	 */
	private List<Map<String, Object>> readExcelValue(Workbook wb) {
		// 得到第一个shell
		Sheet sheet = wb.getSheetAt(0);
		// 得到Excel的行数
		this.totalRows = sheet.getPhysicalNumberOfRows();
		// this.totalRows = sheet.getLastRowNum() + 1;
		// 得到Excel的列数(前提是有行数)
		if (totalRows > 1 && sheet.getRow(0) != null) {
			this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();
		}
		List<Map<String, Object>> userList = new ArrayList<Map<String, Object>>();
		// 循环Excel行数
		for (int r = 1; r < totalRows; r++) {
			Row row = sheet.getRow(r);
			if (row == null) {
				continue;
			}
			// 循环Excel的列
			Map<String, Object> map = new HashMap<String, Object>();
			for (int c = 0; c < this.totalCells; c++) {
				Cell cell = (Cell) row.getCell(c);
				if (null != cell) {
					if (c == 0) {
						// 如果是纯数字,比如你写的是25,cell.getNumericCellValue()获得是25.0,通过截取字符串去掉.0获得25
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String xjh = String.valueOf(cell.getNumericCellValue());
							map.put("0", xjh.substring(0, xjh.length() - 2 > 0 ? xjh.length() - 2 : 1));// 名称
						} else {
							map.put("0", cell.getStringCellValue());// 名称
						}
					} else if (c == 1) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String xm = String.valueOf(cell.getNumericCellValue());
							map.put("1", xm.substring(0, xm.length() - 2 > 0 ? xm.length() - 2 : 1));// 性别
						} else {
							map.put("1", cell.getStringCellValue());// 性别
						}
					} else if (c == 2) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String sfzh = String.valueOf(cell.getNumericCellValue());
							map.put("2", sfzh.substring(0, sfzh.length() - 2 > 0 ? sfzh.length() - 2 : 1));// 年龄
						} else {
							map.put("2", cell.getStringCellValue());// 年龄
						}
					} else if (c == 3) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String bjmc = String.valueOf(cell.getNumericCellValue());
							map.put("3", bjmc.substring(0, bjmc.length() - 2 > 0 ? bjmc.length() - 2 : 1));// 年龄
						} else {
							map.put("3", cell.getStringCellValue());// 年龄
						}
					} else if (c == 4) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String xb = String.valueOf(cell.getNumericCellValue());
							map.put("4", xb.substring(0, xb.length() - 2 > 0 ? xb.length() - 2 : 1));// 年龄
						} else {
							map.put("4", cell.getStringCellValue());// 年龄
						}
					} else if (c == 5) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String mz = String.valueOf(cell.getNumericCellValue());
							map.put("5", mz.substring(0, mz.length() - 2 > 0 ? mz.length() - 2 : 1));// 年龄
						} else {
							map.put("5", cell.getStringCellValue());// 年龄
						}
					} else if (c == 6) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String csny = String.valueOf(cell.getNumericCellValue());
							map.put("6", csny.substring(0, csny.length() - 2 > 0 ? csny.length() - 2 : 1));// 年龄
						} else {
							map.put("6", cell.getStringCellValue());// 年龄
						}
					} else if (c == 7) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String xszt = String.valueOf(cell.getNumericCellValue());
							map.put("7", xszt.substring(0, xszt.length() - 2 > 0 ? xszt.length() - 2 : 1));// 年龄
						} else {
							map.put("7", cell.getStringCellValue());// 年龄
						}
					} else if (c == 8) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String wtlx = String.valueOf(cell.getNumericCellValue());
							map.put("8", wtlx.substring(0, wtlx.length() - 2 > 0 ? wtlx.length() - 2 : 1));// 年龄
						} else {
							map.put("8", cell.getStringCellValue());// 年龄
						}
					} else if (c == 9) {
						if (cell.getCellType() == HSSFCell.CELL_TYPE_NUMERIC) {
							String cwlx = String.valueOf(cell.getNumericCellValue());
							map.put("9", cwlx.substring(0, cwlx.length() - 2 > 0 ? cwlx.length() - 2 : 1));// 年龄
						} else {
							map.put("9", cell.getStringCellValue());// 年龄
						}
					} else {
						log.warn("有更多的字段未能读取保存，请检查数据。。。");
					}
				}
			}
			// 添加到list
			userList.add(map);
		}
		return userList;
	}

	/**
	 * 验证EXCEL文件
	 * 
	 * @param filePath
	 * @return
	 */
	public boolean validateExcel(String filePath) {
		if (filePath == null || !(isExcel2003(filePath) || isExcel2007(filePath))) {
			errorMsg = "文件名不是excel格式";
			return false;
		}
		return true;
	}

	// @描述：是否是2003的excel，返回true是2003
	public static boolean isExcel2003(String filePath) {
		return filePath.matches("^.+\\.(?i)(xls)$");
	}

	// @描述：是否是2007的excel，返回true是2007
	public static boolean isExcel2007(String filePath) {
		return filePath.matches("^.+\\.(?i)(xlsx)$");
	}

}